FROM pytorch/pytorch:latest

RUN mkdir -p /data/app \
    && mkdir -p /root/nltk_data/corpora \
    && apt-get update \
    && apt-get install zip unzip \
    && apt-get -y install vim \
    && apt-get -y install wget \
    && apt-get -y install python3-scipy \
    && pip3 install scipy  -i https://pypi.tuna.tsinghua.edu.cn/simple \
    && pip3 install nltk -i https://pypi.tuna.tsinghua.edu.cn/simple


WORKDIR /data/app

ADD ./hybridAttention ./hybridAttention
ADD ./stopwords /root/nltk_data/corpora/stopwords
ADD data.zip ./

RUN unzip data.zip

